package com.zzyl.common.ai;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.client.RestTemplate;

public class BaiduTest {
    private static final String prompt = "午餐肉能不能晚上吃，那为什么叫午餐肉";

    @Autowired
    private RestTemplate restTemplate;

    public static void main(String[] args) {
        /**
         * 第一个参数：认证类型，固定选择 Auth.TYPE_OAUTH
         * 第二个参数：accessKeyId，从百度云控制台创建的应用里可以找到
         * 第三个参数：accessKeySecret，从百度云控制台创建的应用里可以找到
         */
//        Qianfan qianfan = new Qianfan(Auth.TYPE_OAUTH, "ANsQrJLn85Ae0ptM9qIvbdRg", "6rIzcyg6C0W2aGJgthMSq2RFwPs1ToJl");
//        ChatResponse response = qianfan.chatCompletion()
//                .model("ernie-lite-8k") // 模型名称，要选择自己开通付费的模型
//                .addMessage("user", prompt) // 聊天内容，可以设置多个，每个消息包含role（角色，user表示用户，assistant表示模型），content（消息内容）
//                .temperature(0.1) // 采样参数，取值范围(0,1]
//                .maxOutputTokens(2000) // 模型输出最大长度，取值范围[2, 2048]
//                .responseFormat("json_object")  // 模型输出格式，取值范围：text（文本）、json_object（JSON对象）
//                .execute();
//        String result = response.getResult();
//        System.out.println(result);
    }
}
